Fix kg_emb broken import: SampleBaseDataset -> SampleDataset, add pan… - #1192

Thanks for this. The ImportError diagnosis is right, and the failure chain in
your description, SampleBaseDataset first and then pandarallel, matches what
I reproduced exactly.

I think one step is missing though. SampleDataset in 2.0 isn't the successor
to SampleBaseDataset: it's a litdata.StreamingDataset that expects a
directory containing schema.pkl. SampleKGDataset.__init__ still passes a
list, so:

ValueError: dir_path must be either a string, Path, or Dir, got: <class 'list'>

The module imports, but set_task() can't return anything usable. The three
tests here don't catch it because none of them instantiates the class.

Two smaller notes:

• pandarallel: initialize() is indeed called in umls.py, but the import in
  base_kg_dataset.py, the one cited in the description, is dead, and there is
  no parallel_apply anywhere in kg_emb. It is also not declared in
  pyproject.toml on master, which is what produced the ModuleNotFoundError
  in the first place.
• The test comparing TransE.__init__'s annotation to SampleDataset locks in
  an incorrect hint: KGEBaseModel reads dataset.entity_num, which only
  exists on SampleKGDataset.

Worth saying that #952 asked for the rename across eight files and your PR does
exactly that, so this is really about how the issue was framed.

I've opened #1202 going after the root cause, and credited you in the
description. Would be glad to have your review on it.